package com.dmap.userportrait.config;

import com.fasterxml.jackson.annotation.JsonAutoDetect;
import com.fasterxml.jackson.annotation.PropertyAccessor;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.springframework.cache.CacheManager;
import org.springframework.cache.annotation.EnableCaching;
import org.springframework.cache.interceptor.KeyGenerator;
import org.springframework.cache.jcache.config.JCacheConfigurerSupport;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.cache.RedisCacheManager;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.serializer.Jackson2JsonRedisSerializer;

import java.lang.reflect.Method;

/**
 * Created by tao.liu on 17-7-20.
 */
@SuppressWarnings("Duplicates")
@Configuration
@EnableCaching
public class RedisConfig extends JCacheConfigurerSupport {
    @Bean
    public KeyGenerator redisKeyGenerator() {
        return (target, method, params) -> {
            StringBuilder result = new StringBuilder();
            result.append(target.getClass().getSimpleName() + "[" + target.hashCode() + "]");
            result.append("." + method.getName());
            // 有参数的情况
            if (params != null && params.length > 0) {
                result.append("(");
                for (Object p : params) {
                    result.append(p.toString() + ",");
                }
                result.append(")");
                return result.toString();
            }
            // 没有参数的情况
            result.append("()");
            return result.toString();
        };
    }

    @Bean
    public KeyGenerator simpleKey() {
        return (target, method, params) -> {
            StringBuilder sb = new StringBuilder();
            sb.append(target.getClass().getName() + ":");
            for (Object obj : params) {
                sb.append(obj.toString());
            }
            return sb;
        };
    }


    @Bean
    public CacheManager cacheManager(@SuppressWarnings("rawtypes") RedisTemplate redisTemplate) {
        RedisCacheManager cacheManager = new RedisCacheManager(redisTemplate);
        cacheManager.setDefaultExpiration(60);
        return cacheManager;
    }

    @Bean
    public RedisTemplate<String, String> redisTemplate(RedisConnectionFactory factory) {
        StringRedisTemplate template = new StringRedisTemplate(factory);
        Jackson2JsonRedisSerializer jackson2JsonRedisSerializer = new Jackson2JsonRedisSerializer(Object.class);
        ObjectMapper om = new ObjectMapper();
        om.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY);
        om.enableDefaultTyping(ObjectMapper.DefaultTyping.NON_FINAL);
        jackson2JsonRedisSerializer.setObjectMapper(om);
        template.setValueSerializer(jackson2JsonRedisSerializer);
        template.afterPropertiesSet();
        return template;
    }

}
